package licode.unisop.single.platform.infrastructure.utils;

import com.baomidou.mybatisplus.core.metadata.IPage;
import licode.unisop.provider.vo.SiPageResult;

import java.util.List;
import java.util.function.Function;

/**
 * 分页相关工具类
 *
 * @author licode
 */
public class PageUtil {
    public static <K, T> SiPageResult<K> convertPage(IPage<T> page, Function<List<T>, List<K>> call) {
        SiPageResult<K> pageData = new SiPageResult<>();
        if (page != null) {
            pageData.setCurrent(page.getCurrent());
            pageData.setPageSize(page.getSize());
            pageData.setTotal(page.getTotal());
            pageData.setCount(page.getPages());
            pageData.setData(call.apply(page.getRecords()));
        }
        return pageData;
    }

    public static <K, T> SiPageResult<K> convertPage(SiPageResult<T> page, Function<List<T>, List<K>> call) {
        SiPageResult<K> pageData = new SiPageResult<>();
        if (page != null) {
            pageData.setCurrent(page.getCurrent());
            pageData.setPageSize(page.getPageSize());
            pageData.setTotal(page.getTotal());
            pageData.setCount(page.getCount());
            pageData.setData(call.apply(page.getData()));
        }
        return pageData;
    }
}
